import openpyxl

from config.config import EXCEL_FILE, SHEET_NAME


def read_excel():
    # 打开 excel 文件
    workbook = openpyxl.load_workbook(EXCEL_FILE)   # 参数传文件路径

    # 选择表
    worksheet = workbook[SHEET_NAME]

    data = [] # 空列表, 用于组装字典

    keys = [cell.value for cell in worksheet[2]]    # 拿key行, 也就是表的第二行, 生成一个 key 的列表
    for row in worksheet.iter_rows(min_row=3, values_only=True):    # 从第三行开始拿, 只返回值
        dict_data = dict(zip(keys, row))
        # 如果读取的 is_true 字段的值是 TRUE, 则 append, 否则, 不append
        if dict_data["is_true"]:
            data.append(dict_data)

    # 关闭 excel 文件
    workbook.close()

    return data


